package com.perfma.xtest.cmd;

import org.apache.commons.exec.CommandLine;
import org.apache.commons.exec.DefaultExecuteResultHandler;
import org.apache.commons.exec.DefaultExecutor;
import org.apache.commons.exec.PumpStreamHandler;
import org.apache.commons.io.IOUtils;
import org.voovan.tools.TFile;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;

/**
 * 1、内嵌编译器如"PythonInterpreter"无法引用扩展包，因此推荐使用java调用控制台进程方式"Runtime.getRuntime().exec()"来运行脚本(shell或python)；
 * 2、因为通过java调用控制台进程方式实现，需要保证目标机器PATH路径正确配置对应编译器；
 * 3、暂时脚本执行日志只能在脚本执行结束后一次性获取，无法保证实时性；因此为确保日志实时性，可改为将脚本打印的日志存储在指定的日志文件上；
 * 4、python 异常输出优先级高于标准输出，体现在Log文件中，因此推荐通过logging方式打日志保持和异常信息一致；否则用prinf日志顺序会错乱
 */
public class ScriptUtil {

    public static final String GLUESOURCE_PATH = new File(System.getProperty("java.io.tmpdir"), "gluesource").getAbsolutePath();

    /**
     * make script file
     *
     * @param scriptFileName
     * @param content
     * @throws IOException
     */
    public static void markScriptFile(String scriptFileName, String content) throws IOException {
        // filePath/gluesource/
        File filePathSourceDir = new File(scriptFileName).getParentFile();
        if (!filePathSourceDir.exists()) {
            filePathSourceDir.mkdirs();
        }

        // make file,   filePath/gluesource/666-123456789.py
        FileOutputStream fileOutputStream = null;
        try {
            fileOutputStream = new FileOutputStream(scriptFileName);
            fileOutputStream.write(content.getBytes(StandardCharsets.UTF_8));
            fileOutputStream.close();
        } catch (Exception e) {
            throw e;
        } finally {
            if (fileOutputStream != null) {
                fileOutputStream.close();
            }
        }
    }

    /**
     * 日志文件输出方式
     * <p>
     * 优点：支持将目标数据实时输出到指定日志文件中去
     * 缺点：
     * 标准输出和错误输出优先级固定，可能和脚本中顺序不一致
     * Java无法实时获取
     *
     * @param map        环境信息
     * @param command
     * @param scriptFile
     * @param logFile
     * @return
     * @throws IOException
     */
    public static int execToFile(Map map, String command, String scriptFile, String logFile, Long jobId) throws Exception {
        // 标准输出：print （null if watchdog timeout）
        // 错误输出：logging + 异常 （still exists if watchdog timeout）
        // 标准输入
        File file = new File(logFile);
        if (file.exists()) {
            TFile.deleteFile(file);
        }
        FileOutputStream fileOutputStream = new FileOutputStream(logFile, true);
        PumpStreamHandler streamHandler = new PumpStreamHandler(fileOutputStream, fileOutputStream, null);

        // command
        CommandLine commandline = new CommandLine(command);
        commandline.addArgument(scriptFile);

        DefaultExecuteResultHandler resultHandler = new DefaultExecuteResultHandler();
        // exec
        DefaultExecutor exec = new DefaultExecutor();
        exec.setExitValues(null);
        exec.setStreamHandler(streamHandler);
        String httpFileName = JobFileAppender.makeTagLogFileName(jobId);
        File httpFile = new File(JobFileAppender.logPath, httpFileName);
        File logFilePath = new File(logFile);
        if (map == null) {
            map = new HashMap();
            map.put("JAVA_OPTS=-DtagName", httpFile.getAbsolutePath());
            map.put("logName", logFilePath.getAbsolutePath());
        } else {
            map.put("JAVA_OPTS=-DtagName", httpFile.getAbsolutePath());
            map.put("logName", logFilePath.getAbsolutePath());
        }
        map.put("LANG", "en_US.UTF-8");
        if (XShelter.RUN_IN_XSHELTER) {
            map.put("LANG", "zh_CN.UTF-8");
        }
        exec.execute(commandline, map, resultHandler);
        boolean needBreak=true;
        Long startTime=System.currentTimeMillis();
        Long endTime=System.currentTimeMillis();
        while (needBreak&&endTime-startTime<=30000){
            try{
                resultHandler.waitFor();
                needBreak=false;
            }catch (Exception ex){
                needBreak=true;
            }
            endTime=System.currentTimeMillis();
        }

        IOUtils.closeQuietly(fileOutputStream);
        return resultHandler.getExitValue();
    }

}
